import React from 'react'

export type SizeType = 'small' | 'middle' | 'large' | undefined

const SizeContext = React.createContext<SizeType>(undefined)

export interface SizeContextProps {
  size?: SizeType
}

export const SizeContextProvider: React.FC<SizeContextProps> = ({ children, size }) => (
  <SizeContext.Consumer>
    {(originSize: SizeType) => <SizeContext.Provider value={size || originSize}>{children}</SizeContext.Provider>}
  </SizeContext.Consumer>
)

export default SizeContext
